package com.woniu.aimd.controller.editorAi;

import cn.dev33.satoken.util.SaResult;
import com.woniu.aimd.pojo.DTO.MessageDTO;
import com.woniu.aimd.prompt.AiDialogPrompt;
import jakarta.annotation.Resource;
import org.springframework.ai.chat.client.ChatClient;
import org.springframework.ai.chat.prompt.Prompt;
import org.springframework.ai.chat.prompt.PromptTemplate;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.Map;

/**
 * 编辑器中的ai
 */
@RestController
@RequestMapping("editorAi")
public class EditorAIController {
    @Resource
    @Qualifier("insertChatClient")
    private ChatClient insertChatClient;

    /**
     * 润色
     * @param message
     * @return
     * ruirui
     */
    @RequestMapping("polish")
    public SaResult polish(@RequestBody MessageDTO message){

        return SaResult.ok().setData(insertChatClient.prompt(
                        new PromptTemplate(AiDialogPrompt.polish_CONTENT_PROMPT)
                                .create(Map.of("content", message.getContent())))
                .call().content()) ;
    }

}
